# encoding: utf-8

from lsptrain.nlp_classification_predictor import TextClassificationPredictor

predictor = TextClassificationPredictor(label_file_name="catalog_label.txt",
                                        model_file_name="txt_classification_model_best.pt")

while True:
    text = input("Enter text:")
    result = predictor.predict(string=text)
    print(result)
